vocab_len = 411
embedding_size = 128
lstm_hidden_size = 256
lstm_layer_nums = 2
lstm_bidirectional = False
drop_out = 0.2
# is_parent = True
is_parent = False
if is_parent:
    n_classes = 5
    model_path = 'parent_model.pkl'
    data_path = 'parent_data.csv'
else:
    n_classes = 15
    model_path = 'child_model.pkl'
    data_path = 'child_data.csv'

epochs = 800
lr = 0.001
batch_size = 256
